from selenium import webdriver
driver = webdriver.Chrome()
driver.get("https://www.baidu.com")

# 第一种截图的方式：直接截图保存在本地 传递一个参数作为图片的名字
driver.save_screenshot("test.png")
# 第二种截图的方式： 将截图保存为本地文件，后缀是png的格式 最终的效果是跟上面的一致的
driver.get_screenshot_as_file("test1.png")
# 第三种截图方法： 将截图返回为二进制的数据  不需要参数 == 用的不多
# print(driver.get_screenshot_as_png())
# 第四种截图的方式： 将截图返回baser64编码的数据  不需要参数 == 用的不多
print(driver.get_screenshot_as_base64())
